#pragma once

#include "compiler.hpp"
#include "runner.hpp"
#include "../comm/log.hpp"
#include "../comm/util.hpp"

#include <signal.h>
#include <unistd.h>
#include <jsoncpp/json/json.h>

namespace ns_complie_run
{
    using namespace ns_compiler;
    using namespace ns_runner;
    using namespace ns_log;
    using namespace ns_util;

    class Compile_Run
    {
        // 将code转换为resaon
    public:
        //清理临时文件 unlink(path) 清除指定文件
        static void RemoveTempFile(const std::string& file_name)
        {
            std::string src=PathUtil::Src(file_name);
            if(FileUtil::IsFileExist(src)) unlink(src.c_str());

            std::string compile_err=PathUtil::CompilerError(file_name);
            if(FileUtil::IsFileExist(compile_err)) unlink(compile_err.c_str());

            std::string exe=PathUtil::Exe(file_name);
            if(FileUtil::IsFileExist(exe)) unlink(exe.c_str());

            std::string stdin=PathUtil::Stdin(file_name);
            if(FileUtil::IsFileExist(stdin)) unlink(stdin.c_str());

            std::string stdout=PathUtil::Stdout(file_name);
            if(FileUtil::IsFileExist(stdout)) unlink(stdout.c_str());

            std::string stderr=PathUtil::Stderr(file_name);
            if(FileUtil::IsFileExist(stderr)) unlink(stderr.c_str());
        }
        static std::string CodeToDesc(int code, const std::string &file_name)
        {
            std::string desc;
            switch (code)
            {
            case 0:
                desc = "编译成功";
                break;
            case -1:
                desc = "提交代码为空";
                break;
            case -2:
                desc = "未知错误";
                break;
            case -3:
                FileUtil::ReadFile(PathUtil::CompilerError(file_name), &desc, true);
                break;
            case SIGABRT:
                desc = "内存超过范围";
                break;
            case SIGXCPU:
                desc = "cpu运行超时";
                break;
            case SIGFPE:
                desc = "浮点数溢出";
                break;
            default:
                desc = "unknown " + std::to_string(code);
                break;
            }
            return desc;
        }
        /*
        输入    code : 代码
                input : 用户给予的输入
                cpu_limit : cpu限制运行的时间
                mem_limit : 内存限制使用的大小
        输出    status : 状态码
                reason : 请求结果
                stdout : 标准输出
                stderr : 标准错误
        */

        // 使用json进行序列化和反序列化
        static void Start(const std::string &in_json, std::string *out_json)
        {
            // 首先进行反序列化
            Json::Value in_value;
            Json::Reader reader;
            reader.parse(in_json, in_value);

            // 拿到用户发送的数据
            std::string code = in_value["code"].asString();
            std::string input = in_value["input"].asString();
            int cpu_limit = in_value["cpu_limit"].asInt();
            int mem_limit = in_value["mem_limit"].asInt();

            int sum_result = 0;
            int run_result = 0;
            Json::Value out_value;
            std::string file_name;

            if (code.empty())
            {
                sum_result = -1;
                goto END;
            }
            file_name = FileUtil::UniqueFilename();

            // 生成一个src (xxx.cc)文件
            if (!FileUtil::WriteFile(PathUtil::Src(file_name), code))
            {
                sum_result = -2;
                goto END;
            }

            // 进行编译
            if (!Compiler::Compile(file_name))
            {
                sum_result = -3;
                goto END;
            }
            //std::cout << "编译成功，准备runing" << std::endl;
            run_result = Runner::Run(file_name, cpu_limit, mem_limit);

            if (run_result < 0)
            {
                sum_result = -2;
            }
            else if (run_result > 0)
            {
                // 将信号传给处理函数
                sum_result = run_result;
            }
            else // 表示运行已经成功
            {
                sum_result = 0;
            }
        END:

            // 将value的值设置完毕
            out_value["status"] = sum_result;
            out_value["reason"] = CodeToDesc(sum_result, file_name);

            // 非必要内容，当全部不出错在设置输出和错误
            if (sum_result == 0)
            {
                std::string _stdout;
                FileUtil::ReadFile(PathUtil::Stdout(file_name), &_stdout, true);
                out_value["stdout"] = _stdout;

                std::string _stderr;
                FileUtil::ReadFile(PathUtil::Stderr(file_name), &_stderr, true);
                out_value["stderr"] = _stderr;
            }

            Json::StyledWriter writer;
            // 写入out_json
            (*out_json) = writer.write(out_value);

            RemoveTempFile(file_name);
        }
    };
}
